package com.cmyshare.design.factoryPattern;

/**
 * @version 1.0
 * @Author cmy
 * @desc 4、创建一个作业工厂类，用于根据需求创建作业对象：
 */
public class WorkFactory {
    /**
     * 创建作业对象
     *
     * @param type
     * @return
     */
    public Work createWork(String type) {
        if ("Student".equals(type)) {
            // 创建学生对象并返回作业接口的实现类 Student 类型对象。
            return new StudentWork();
        } else if ("Teacher".equals(type)) {
            // 创建老师对象并返回作业接口的实现类 Teacher 类型对象。
            return new TeacherWork();
        } else {
            // 如果类型不匹配，抛出异常。
            throw new IllegalArgumentException("Invalid assignment type.");
        }
    }
}
